"""
案例：
    有login()函数：
    用户名admin，密码123456，可实现成功登录，返回"登录成功"，否则返回"登录失败"。
    使用assert 结合 == 和 in 实现两种情况的断言
"""
# 准备待测函数
def login(uname, pwd):
    if uname == "admin" and pwd == "123456":
        return "登录成功"
    else:
        return "登录失败"


# 定义测试类
class TestLogin(object):
    # 测试方法
    def test1_login_success(self):
        # 调用待测函数
        ret = login("admin", "123456")
        # 断言
        assert "登录成功" == ret

    def test2_login_failed(self):
        # 调用待测函数，获取实际结果
        ret = login("admin", "111111")
        # 断言
        assert "失败" in ret
